import random
import cv2
import numpy as np
import shutil
import DataAugment
#from DataAugmentation import DataAugment

_min_angel         = 60	# for img_rotation
_max_angel         = 60		# for img_rotation
_min_scale         = 1.0	# for img_rotation
_max_scale         = 1.0	# for img_rotation
_fill_pixel        = 255	# for img_shift and img_rotation

img = cv2.imread('3.jpg')
for angle in range(0, 360, 10):
    img1, global_angle, Matrix = DataAugment.img_rotation(img, angle, angle, _min_scale, _max_scale, _fill_pixel)
    cv2.imwrite('image_%d.jpg' % (angle), img1)
    cv2.imshow('image', img1)
    cv2.waitKey(10)